1612G - Max Sum Array - CodeForces Solution


combinatorics constructive algorithms greedy sortings *2500

Please click on ads to support us..

Python Code:

import sys, os, io
input = io.BytesIO(os.read(0, os.fstat(0).st_size)).readline

m = int(input())
mod = pow(10, 9) + 7
c = list(map(int, input().split()))
n = pow(10, 6) + 5
fact = [1] * (n + 1)
for i in range(1, n + 1):
    fact[i] = i * fact[i - 1] % mod
cnt = [0] * (n + 1)
for i in c:
    cnt[i - 1] += 1
for i in range(n - 2, -1, -1):
    cnt[i] += cnt[i + 2]
s, c, u = 0, 1, 0
inv2 = pow(2, mod - 2, mod)
y = 0
for i in range(-n, n):
    j = abs(i)
    if not cnt[j]:
        continue
    u += cnt[j]
    u %= mod
    x = u * (u + 1) % mod
    s += i * (x - y) % mod
    s %= mod
    y = x
    c *= fact[cnt[j]]
    c %= mod
s = s * inv2 % mod
print(s, c)


Comments

Submit
0 Comments
More Questions

1077A - Frog Jumping
1714G - Path Prefixes
1369C - RationalLee
289B - Polo the Penguin and Matrix
1716A - 2-3 Moves
1670B - Dorms War
1716B - Permutation Chain
987A - Infinity Gauntlet
1676G - White-Black Balanced Subtrees
1716D - Chip Move
1352F - Binary String Reconstruction
1487B - Cat Cycle
1679C - Rooks Defenders
56A - Bar
1694B - Paranoid String
35A - Shell Game
1684A - Digit Minimization
43B - Letter
1017A - The Rank
1698B - Rising Sand
235A - LCM Challenge
1075B - Taxi drivers and Lyft
1562A - The Miracle and the Sleeper
1216A - Prefixes
1490C - Sum of Cubes
868A - Bark to Unlock
873B - Balanced Substring
1401D - Maximum Distributed Tree
1716C - Robot in a Hallway
1688B - Patchouli's Magical Talisman